package com.tensquare.base.service;

import com.tensquare.base.dao.LabelDao;
import com.tensquare.base.pojo.Label;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import util.IdWorker;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName: LabelService
 * @Description: //TODO
 * @Author: LL
 * @Date: 2019/6/1 15:29
 * @Version: V1.0
 */
@Service
@Transactional
public class LabelService {

    @Autowired
    private LabelDao labelDao;
    @Autowired
    private IdWorker idWorker;

    //查询全部
    public List<Label> findAll() {
        return labelDao.findAll();
    }

    //根据id查询
    public Label findById(String id) {
        return labelDao.findById(id).get();
    }

    //添加
    public void save(Label label) {
        label.setId(idWorker.nextId() + "");
        labelDao.save(label);
    }

    //更新  save()既可以做保存又可以做更新,先根据id进行查询，查询到了就进行更新，没查询到就进行保存
    public void update(Label label) {
        labelDao.save(label);
    }

    //删除
    public void deleteById(String id) {
        labelDao.deleteById(id);
    }

    //条件查询
    public List<Label> findSearch(Label label) {
        return labelDao.findAll(new Specification<Label>() {

            /*
             * @Param: root   root根对象，也就是要把条件封装到哪个对象中，where类名 = label.getId
             * @Param: query  query封装查询关键字，比如：group by, order by等
             * @Param: cb     cd用来封装条件对象，如果直接返回null，表示不需要任何条件
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                //new一个list集合，用来存放所有条件
                List<Predicate> list = new ArrayList<>();
                if (!StringUtils.isEmpty(label.getLabelname())) {
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");//等于：where labelname like "%小明%"
                    list.add(predicate);
                }
                if (!StringUtils.isEmpty(label.getState())) {
                    Predicate predicate = cb.equal(root.get("state").as(String.class), label.getState());// state="1"
                    list.add(predicate);
                }
                //new一个数组作为最终返回值的条件
                Predicate[] parr = new Predicate[list.size()];
                //把list转化成数组
                list.toArray(parr);
                return cb.and(parr); //相当于 where labelname like "%小明%" and state="1"
            }
        });
    }

    //条件、分页查询
    public Page<Label> pageQuery(Label label, int page, int size) {

        // 封装分页对象
        Pageable pageable = PageRequest.of(page - 1, size);
        return labelDao.findAll(new Specification<Label>() {

            /*
             * @Param: root   root根对象，也就是要把条件封装到哪个对象中，where类名 = label.getId
             * @Param: query  query封装查询关键字，比如：group by, order by等
             * @Param: cb     cd用来封装条件对象，如果直接返回null，表示不需要任何条件
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                //new一个list集合，用来存放所有条件
                List<Predicate> list = new ArrayList<>();
                if (!StringUtils.isEmpty(label.getLabelname())) {
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");//等于：where labelname like "%小明%"
                    list.add(predicate);
                }
                if (!StringUtils.isEmpty(label.getState())) {
                    Predicate predicate = cb.equal(root.get("state").as(String.class), label.getState());// state="1"
                    list.add(predicate);
                }
                //new一个数组作为最终返回值的条件
                Predicate[] parr = new Predicate[list.size()];
                //把list转化成数组
                list.toArray(parr);
                return cb.and(parr); //相当于 where labelname like "%小明%" and state="1"
            }
        }, pageable);
    }
}
